from literalai import LiteralClient

literalai_client = LiteralClient(
    url="http://localhost:3000", api_key="my-initial-api-key"
)

def get_prompt():
    prompt = literalai_client.api.get_prompt(name="Default", version=2)
    prompt2 = literalai_client.api.get_or_create_prompt(
        name="Default", template_messages=[]
    )
    prompt_lineage = literalai_client.api.get_or_create_prompt_lineage(name="Default")
    print("prompt", prompt)
    print("prompt2", prompt2)
    print("prompt lineage", prompt_lineage)

if __name__ == "__main__":
    get_prompt()